#!/bin/bash

# 监控服务卸载脚本
# 用于卸载IO监控服务及相关文件

# 颜色定义
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# 日志文件
LOG_FILE="/var/log/io-monitor-uninstall.log"
exec > >(tee -a "$LOG_FILE") 2>&1

# 检查是否以root权限运行
if [[ $EUID -ne 0 ]]; then
   echo -e "${RED}此脚本需要root权限运行${NC}"
   echo "请使用 sudo $0 重新运行"
   exit 1
fi

echo -e "${BLUE}===== IO监控服务卸载 =====${NC}"
echo "此脚本将卸载IO监控服务及相关文件"

# 停止并禁用服务
echo -e "${BLUE}[1/4] 停止并禁用服务...${NC}"
if systemctl stop io-monitor.service 2>/dev/null; then
    echo -e "${GREEN}✓ 服务已停止${NC}"
else
    echo -e "${YELLOW}⚠ 服务停止失败，可能未运行或不存在${NC}"
fi

if systemctl disable io-monitor.service 2>/dev/null; then
    echo -e "${GREEN}✓ 服务已禁用${NC}"
else
    echo -e "${YELLOW}⚠ 服务禁用失败，可能未启用或不存在${NC}"
fi

# 删除systemd服务文件
echo -e "${BLUE}[2/4] 删除systemd服务文件...${NC}"
SERVICE_FILE="/etc/systemd/system/io-monitor.service"
if [ -f "$SERVICE_FILE" ]; then
    rm -f "$SERVICE_FILE"
    echo -e "${GREEN}✓ systemd服务文件已删除${NC}"
else
    echo -e "${YELLOW}⚠ systemd服务文件不存在${NC}"
fi

# 重新加载systemd配置
echo -e "${BLUE}[3/4] 重新加载systemd配置...${NC}"
if systemctl daemon-reload 2>/dev/null; then
    echo -e "${GREEN}✓ systemd配置已重新加载${NC}"
else
    echo -e "${YELLOW}⚠ systemd配置重新加载失败${NC}"
fi

# 删除系统分区中的脚本
echo -e "${BLUE}[4/4] 删除系统分区中的脚本...${NC}"
SCRIPT_DIR="/opt/ohmylinux/io-monitor"
if [ -d "$SCRIPT_DIR" ]; then
    rm -rf "$SCRIPT_DIR"
    echo -e "${GREEN}✓ 系统分区中的脚本已删除${NC}"
else
    echo -e "${YELLOW}⚠ 系统分区中的脚本不存在${NC}"
fi

echo
echo -e "${GREEN}===== IO监控服务卸载完成 =====${NC}"
echo "IO监控服务已完全从系统中移除"